package com.ht.modules.system.controller;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.List;
import java.util.Map;

import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;

import org.apache.shiro.SecurityUtils;
import com.ht.entity.Result;
import com.ht.aspect.annotation.AutoLog;
import com.ht.query.QueryGenerator;
import com.ht.entity.LoginUser;
import com.ht.utils.oConvertUtils;
import com.ht.modules.system.entity.DataRule;
import com.ht.modules.system.entity.DataRuleVo;
import com.ht.modules.system.service.IDataRuleService;
import com.ht.modules.system.util.DataRuleUtil;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;

import com.alibaba.fastjson2.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;

 /**
 *  数据规则
 * @Author:
 * @Date:   2021-08-13
 * @Version: V1.0
 */
@Slf4j
@Api(tags="数据规则")
@RestController
@RequestMapping("/datarule/dataRule")
public class DataRuleController {

	@Autowired
	private IDataRuleService dataRuleService;

	/**
	 * 分页列表查询
	 * @param dataRuleVo
	 * @param pageNo 页码
	 * @param pageSize 当页大小
	 * @param column 默认排序列
	 * @param order 默认排序方式
	 * @return
	 */
	@ApiOperation(value="数据规则-分页列表查询", notes="数据规则-分页列表查询")
	@GetMapping(value = "/list")
	public Result<IPage<DataRuleVo>> queryPageList(DataRuleVo dataRuleVo,
									  @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
									  @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
									  @RequestParam(name="column",defaultValue="createTime") String column,
                                      @RequestParam(name="order",defaultValue="desc") String order,HttpServletRequest request) {
		Result<IPage<DataRuleVo>> result = new Result<>();
		Page<DataRuleVo> page = new Page<>(pageNo, pageSize);
		DataRuleUtil.getSql(request);
		page = dataRuleService.list(page, dataRuleVo, column, order);
		result.success("查询成功！");
		result.setResult(page);
		return result;
	}

	/**
	 * 添加
	 * @param dataRule
	 * @return
	 */
	@AutoLog(value = "数据规则-添加")
	@ApiOperation(value="数据规则-添加", notes="数据规则-添加")
	@PostMapping(value = "/add")
	public Result<?> add(@RequestBody DataRule dataRule) {
		Result<?> result = new Result<>();
		try {
			dataRuleService.add(dataRule);
			result.success("添加成功！");
		} catch (Exception e) {
			log.error(e.getMessage(),e);
			result.error500("操作失败！" + e.getMessage());
		}
		return result;
	}

	/**
	 * 编辑
	 * @param dataRule
	 * @return
	 */
	@AutoLog(value = "数据规则-编辑")
	@ApiOperation(value="数据规则-编辑", notes="数据规则-编辑")
	@PutMapping(value = "/edit")
	public Result<?> edit(@RequestBody DataRule dataRule) {
		Result<?> result = new Result<>();
		try {
            dataRuleService.edit(dataRule);
            result.success("修改成功！");
        } catch (Exception e) {
            log.error(e.getMessage(),e);
            result.error500("操作失败！" + e.getMessage());
        }
		return result;
	}

	/**
	 * 通过id删除（逻辑删除）
	 * @param id
	 * @return
	 */
	@AutoLog(value = "数据规则-通过id删除")
	@ApiOperation(value="数据规则-通过id删除", notes="数据规则-通过id删除")
	@DeleteMapping(value = "/delete")
	public Result<?> delete(@RequestParam(name="id") String id) {
		Result<?> result = new Result<>();
		try {
            dataRuleService.delete(id);
            result.success("删除成功！");
        } catch (Exception e) {
            log.error(e.getMessage(),e);
            result.error500("操作失败！" + e.getMessage());
        }
		return result;
	}

	/**
	 * 批量删除（逻辑删除）
	 * @param ids
	 * @return
	 */
	@AutoLog(value = "数据规则-批量删除")
	@ApiOperation(value="数据规则-批量删除", notes="数据规则-批量删除")
	@DeleteMapping(value = "/deleteBatch")
	public Result<?> deleteBatch(@RequestParam(name="ids") String ids) {
		Result<?> result = new Result<>();
		try {
            dataRuleService.deleteBatch(ids);
            result.success("批量删除成功！");
        } catch (Exception e) {
            log.error(e.getMessage(),e);
            result.error500("操作失败！" + e.getMessage());
        }
		return result;
	}

	/**
	 * 通过id查询
	 * @param id
	 * @return
	 */
	@ApiOperation(value="数据规则-通过id查询", notes="数据规则-通过id查询")
	@GetMapping(value = "/queryById")
	public Result<DataRule> queryById(@RequestParam(name="id") String id) {
		Result<DataRule> result = new Result<>();
		try {
            DataRule dataRule = dataRuleService.queryById(id);
            result.setResult(dataRule);
            result.success("查询成功！");
        } catch (Exception e) {
            result.error500("操作失败！" + e.getMessage());
        }
		return result;
	}

  /**
   * 导出excel
   *
   * @param request
   * @param response
   */
  @RequestMapping(value = "/exportXls")
  public ModelAndView exportXls(HttpServletRequest request, HttpServletResponse response) {
      // Step.1 组装查询条件
      QueryWrapper<DataRule> queryWrapper = null;
      LoginUser sysUser = (LoginUser)SecurityUtils.getSubject().getPrincipal();
      try {
          String paramsStr = request.getParameter("paramsStr");
          if (oConvertUtils.isNotEmpty(paramsStr)) {
              String deString = URLDecoder.decode(paramsStr, "UTF-8");
              DataRule dataRule = JSON.parseObject(deString, DataRule.class);
              queryWrapper = QueryGenerator.initQueryWrapper(dataRule, request.getParameterMap());
          }
      } catch (UnsupportedEncodingException e) {
          e.printStackTrace();
      }
      //Step.2 AutoPoi 导出Excel
      ModelAndView mv = new ModelAndView(new JeecgEntityExcelView());
      List<DataRule> pageList = dataRuleService.list(queryWrapper);
      //导出文件名称
      mv.addObject(NormalExcelConstants.FILE_NAME, "数据规则列表");
      mv.addObject(NormalExcelConstants.CLASS, DataRule.class);
      mv.addObject(NormalExcelConstants.PARAMS, new ExportParams("数据规则列表数据", "导出人:"+sysUser.getRealname(), "导出信息"));
      mv.addObject(NormalExcelConstants.DATA_LIST, pageList);
      return mv;
  }

  /**
   * 通过excel导入数据
   *
   * @param request
   * @param response
   * @return
   */
  @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
  public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
      MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
      Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
      for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
          MultipartFile file = entity.getValue();// 获取上传文件对象
          ImportParams params = new ImportParams();
          params.setTitleRows(2);
          params.setHeadRows(1);
          params.setNeedSave(true);
          try {
              List<DataRule> listDataRules = ExcelImportUtil.importExcel(file.getInputStream(), DataRule.class, params);
              for (DataRule dataRuleExcel : listDataRules) {
                  dataRuleService.save(dataRuleExcel);
              }
              return Result.ok("文件导入成功！数据行数:" + listDataRules.size());
          } catch (Exception e) {
              log.error(e.getMessage(),e);
              return Result.error("文件导入失败:"+e.getMessage());
          } finally {
              try {
                  file.getInputStream().close();
              } catch (IOException e) {
                  e.printStackTrace();
              }
          }
      }
      return Result.ok("文件导入失败！");
  }

}
